import torch
import torch.nn.functional as F

t4d = torch.zeros(1, 3, 5, 3)

import torch
import torch.nn.functional as F

t4d = torch.zeros(1, 3, 5, 3)
p1d = (1, 2)
t1 = F.pad(t4d, [1, 2, 3, 4])

print(t1)
print(t1.shape)
